package com.hmdp.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.hmdp.dto.Result;
import com.hmdp.entity.VoucherOrder;
import com.hmdp.mapper.VoucherOrderMapper;
import com.hmdp.service.ISeckillVoucherService;
import com.hmdp.service.IVoucherOrderService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hmdp.utils.RedisIdWorker;
import com.hmdp.utils.UserHolder;
import lombok.extern.slf4j.Slf4j;
import lombok.val;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.aop.framework.AopContext;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.connection.stream.*;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.time.Duration;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.concurrent.*;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 聪哥
 * @since 2024-09-13
 */
@Service
@Slf4j
public class VoucherOrderServiceImpl extends ServiceImpl<VoucherOrderMapper, VoucherOrder> implements IVoucherOrderService {

    @Resource
    private ISeckillVoucherService seckillVoucherService;

    /**
     * Redis操作对象
     */
    @Resource
    private StringRedisTemplate stringRedisTemplate;
    /**
     * 全局ID生成器
     */
    @Resource
    private RedisIdWorker redisIdWorker;

    /**
     * Redisson锁对象
     */
    @Resource
    private RedissonClient redissonClient;

/*-----------------------------第一种方案:优化了好几次的分布式锁方案 最终优化成Redisson提供的锁方案-------------------------------------------------------------*/

    /**
     * 秒杀优惠券下单
     * @param voucherId
     * @return
     */
//    @Override
//    public Result seckillVoucher(Long voucherId) {
//        //1. 提交优惠卷id 查询优惠卷id信息
//        SeckillVoucher voucher = seckillVoucherService.getById(voucherId);
//        if(voucher == null){
//            return Result.fail("优惠券不存在!");
//        }
//
//        //2. 判断优惠卷是否在抢购时间内
//        LocalDateTime beginTime = voucher.getBeginTime();
//        LocalDateTime endTime = voucher.getEndTime();
//        long stockCount =voucher.getStock();
//        if(beginTime.isAfter(LocalDateTime.now())){
//            //   否----> 返回异常信息---->结束
//            return Result.fail("秒杀尚未开始!");
//        }
//        if(endTime.isBefore(LocalDateTime.now())){
//            //   否----> 返回异常信息---->结束
//            return Result.fail("秒杀已经结束!");
//        }
//
//        //3. 判断优惠卷库存是否充足
//        if(stockCount <= 0){
//            //   否----> 返回异常信息---->结束
//            return Result.fail("库存不足!");
//        }
//
//
//        /* 为什么要加在createVoucherOrder方法中？
//         * 因为createVoucherOrder方法中已经加了事务，如果加了锁，事务回滚，锁也会释放，这样就会导致锁失效
//
//         * 为什么要指定锁用户id？
//         * 将锁的粒度降低，将锁的粒度降低到用户粒度，这样就可以保证一个用户只会有一个订单
//         * 添加.intern() 确保是按值加锁  而不是按对象加锁
//
//         * 为什么要使用代理对象？
//         * @Transactional注解是Spring的事务注解，它只能在Spring管理的Bean中生效，
//         * 如果直接在ServiceImpl中调用createVoucherOrder方法，
//         * 那么@Transactional注解就不会生效，
//         * 因为createVoucherOrder方法不是在Spring管理的Bean中调用的，
//         * 所以需要使用代理对象来调用createVoucherOrder方法，这样@Transactional注解就会生效
//         */
//        Long userId = UserHolder.getUser().getId();
//
//        /*
//         *   分布式锁方案
//         */
////        // 1. 创建锁对象
////        SimpleRedisLock lock = new SimpleRedisLock("order:" + userId, stringRedisTemplate);
////        // 2. 尝试获取锁
////         boolean isLock =  lock.tryLock(1200);
//
//
//        /*
//         *   Redisson方案
//         */
//        //1. 创建锁对象
//        RLock lock =  redissonClient.getLock("lock:order:" + userId);
//
//        //2. 尝试获取锁
//        boolean isLock = lock.tryLock();
//
//        // 3. 判断锁是否获取成功
//        if(! isLock){
//            return Result.fail("不允许重复下单");
//        }
//        try {
//            IVoucherOrderService proxy = (IVoucherOrderService) AopContext.currentProxy();
//            return proxy.createVoucherOrder(voucherId);
//        } finally {
//            // 4. 释放锁
//            lock.unlock();
//        }
//
//    }

//    @Transactional
//    public Result createVoucherOrder(Long voucherId) {
//        //4. 限制一人一单【悲观锁方案】
//        Long userId = UserHolder.getUser().getId();
//
//        //4.1 查询订单
//        int count = query().eq("user_id", userId).eq("voucher_id", voucherId).count();
//        //4.2 判断订单是否存在
//        // 是 -----> 返回异常信息---->结束
//        if (count > 0) {
//            return Result.fail("该用户已经购买过一次了！");
//        }
//
//        //5. 扣减库存——解决超卖问题【乐观锁方案】
//        boolean success = seckillVoucherService.update()
//                .setSql("stock = stock-1")
//                .eq("voucher_id", voucherId).gt("stock", 0) // 库存大于0就行了
//                .update();
//        if (!success) {
//            return Result.fail("库存不足!");
//        }
//
//        //6. 创建优惠卷订单
//        VoucherOrder voucherOrder = new VoucherOrder();
//        //6.1 订单id 【全局ID生成器】4
//        long orderId = redisIdWorker.nextId("order");
//        voucherOrder.setId(orderId);
//        //6.2 用户id 【当前登录用户】
////        long userId =  UserHolder.getUser().getId();
//        voucherOrder.setUserId(userId);
//        //6.3 优惠卷id 【传递过来的优惠卷id】
//        voucherOrder.setVoucherId(voucherId);
//        save(voucherOrder);
//
//        //7. 返回订单id ----> 结束
//        return Result.ok(orderId);
//    }


/*-----------------------------第一种方案结束-------------------------------------------------------------*/


    /** 方案二、三公共代码
     * 预加载lua脚本
     */
    private static DefaultRedisScript<Long> SECKILL_SCRIPT;
    static {
        SECKILL_SCRIPT = new DefaultRedisScript<>();
        // 这是第二种方案需要执行的lua脚本
        // SECKILL_SCRIPT.setLocation(new ClassPathResource("lua/seckill.lua"));
        // 这是第三种方案需要执行的lua脚本
        SECKILL_SCRIPT.setLocation(new ClassPathResource("lua/streamSeckill.lua"));
        SECKILL_SCRIPT.setResultType(Long.class);

    }

/*-----------------------------第二种方案： 使用java自带的阻塞队列 + redis + lua脚本判断秒杀资格的方案-------------------------------------------------------------*/

    //1. 创建-- 阻塞队列
//    private BlockingQueue<VoucherOrder> orderTasks = new ArrayBlockingQueue<>(1024 * 1024);

    //2，创建-- 秒杀线程池
//    private static final ExecutorService SECKILL_ORDER_EXECUTOR = Executors.newSingleThreadExecutor();

    //5. 初始化方法  一初始化就执行
//    @PostConstruct
//    public void init(){
//        SECKILL_ORDER_EXECUTOR.submit(new VoucherOrderHandler());
//    }

    //4. 创建-- 订单外部类
//    private void handleVocherOrder(VoucherOrder voucherOrder){
//        // 获取用户
//        Long userId = voucherOrder.getUserId();
//
//        // 1. 创建锁对象
//        RLock lock =  redissonClient.getLock("lock:order:" + userId);
//
//        //2. 尝试获取锁
//        boolean isLock = lock.tryLock();
//
//        // 3. 判断锁是否获取成功
//        if(! isLock){
//            log.error("不允许重复下单");
//        }
//        try {
//             proxy.createVoucherOrder(voucherOrder);
//        } finally {
//            // 4. 释放锁
//            lock.unlock();
//        }
//    }

    //3. 创建-- 秒杀线程任务
//    private class VoucherOrderHandler implements Runnable{
//
//        @Override
//        public void run() {
//            while (true) {
//                try {
//                    //1. 获取队列中的订单信息
//                    VoucherOrder voucherOrder = orderTasks.take();
//                    //2. 创建订单
//                    handleVocherOrder(voucherOrder);
//                } catch (Exception e) {
//                    log.error("获取订单异常",e);
//                }
//            }
//        }
//    }

    /**
     * 秒杀优惠券下单------秒杀优化代码----lua脚本---主线程---使用java自带的阻塞队列完成的
     * @param voucherId
     * @return
     */
//    private IVoucherOrderService proxy;
//    @Override
//    public Result seckillVoucher(Long voucherId) {
//        // 获取用户
//        Long userId = UserHolder.getUser().getId();
//
//        //1.执行Lua脚本
//        Long result = stringRedisTemplate.execute(
//                        SECKILL_SCRIPT,
//                        Collections.emptyList(),
//                        voucherId.toString(),userId.toString()
//        );
//        //2.判断结果是否为0
//        int r = result.intValue();
//        if(r != 0){
//            //3.不为0,代表没有购买资格
//            return Result.fail(r == 1 ? "库存不足!" : "不能重复下单!");
//        }
//
//        long orderId =  redisIdWorker.nextId("order");
//
//        //4.为0,代表有购买资格，将下单信息保存至阻塞队列
//        VoucherOrder voucherOrder = new VoucherOrder();
//        voucherOrder.setId(orderId);
//        voucherOrder.setUserId(userId);
//        voucherOrder.setVoucherId(voucherId);
//
//        // 放入阻塞队列
//        orderTasks.add(voucherOrder);
//        //提前 获取代理对象
//        proxy = (IVoucherOrderService) AopContext.currentProxy();
//
//        //5.返回订单id
//        return Result.ok(orderId);
//    }
//

//    /**
//     * 秒杀优惠券下单------秒杀优化代码----创建订单
//     * @param voucherOrder
//     */
//    @Transactional
//    public void createVoucherOrder(VoucherOrder voucherOrder) {
//        //4. 限制一人一单【悲观锁方案】
//        Long userId = voucherOrder.getUserId();
//
//        //4.1 查询订单
//        int count = query().eq("user_id", userId).eq("voucher_id", voucherOrder.getVoucherId()).count();
//        //4.2 判断订单是否存在
//        // 是 -----> 返回异常信息---->结束
//        if (count > 0) {
//            log.error("用户已经购买了一次了");
//        }
//
//        //5. 扣减库存——解决超卖问题【乐观锁方案】
//        boolean success = seckillVoucherService.update()
//                .setSql("stock = stock-1")
//                .eq("voucher_id", voucherOrder.getVoucherId()).gt("stock", 0) // 库存大于0就行了
//                .update();
//        if (!success) {
//            log.error("库存不足");
//        }
//
//        //6. 创建订单
//        save(voucherOrder);
//    }


/*-----------------------------第二种方案结束------------------------------------------------------------*/


/*-----------------------------第三种方案： 使用Redis的stream消息队列 + redis + lua脚本判断秒杀资格添加消息队列 的方案-------------------------------------------------------------*/

    // 1，创建-- 秒杀线程池
    private static final ExecutorService SECKILL_ORDER_EXECUTOR = Executors.newSingleThreadExecutor();

    //2. 初始化方法  一初始化就执行
    @PostConstruct
    public void init(){
        SECKILL_ORDER_EXECUTOR.submit(new VoucherOrderHandler());
    }

    //3. 创建线程任务用于接收消息队列的信息
    private class VoucherOrderHandler implements Runnable{

        // 消息队列名称
        private String queueName = "stream.orders";

        @Override
        public void run() {
            while (true) {
                try{
                    //1. 获取队列中的订单信息 XREADGROUP GROUP g1 c1 COUNT 1 BLOCK 2000 STREAMS stream.oredes >
                    // 指定队列名称，组名称，消费者名称，读取模式，读取数量，阻塞时间，队列名称，读取位置
                    List<MapRecord<String,Object,Object>> list = stringRedisTemplate.opsForStream().read(
                            Consumer.from("g1", "c1"),
                            StreamReadOptions.empty().count(1).block(Duration.ofSeconds(2)),
                            StreamOffset.create(queueName, ReadOffset.lastConsumed())
                    );
                    //2. 判断消息获取是否成功
                    if( list == null || list.isEmpty()){
                        //2.1 获取失败 说明没有消息 ---->继续循环
                        continue;
                    }
                    // 解析消息中的订单信息
                    MapRecord<String,Object,Object> record = list.get(0);
                    //  获取键值对集合
                    Map<Object,Object> values = record.getValue();
                    // 获取订单信息
                    VoucherOrder voucherOrder = BeanUtil.fillBeanWithMap(values, new VoucherOrder(), true);

                    //3. 获取成功，执行订单创建
                    handleVocherOrder(voucherOrder);

                    //4. ACK确认  SACK stream.orders g1 id
                    stringRedisTemplate.opsForStream().acknowledge(queueName,"g1",record.getId());

                }catch (Exception e) {
                    // 消息没有被ACK确认 进入Pending List
                    log.error("订单处理出现异常",e);
                    handlePendingList();
                }
            }
        }


        // 5.取不到订单————— 处理Pending List中的订单信息
        private void handlePendingList(){
            while (true) {
                try {
                    //1. 获取Pending List中的订单信息 XREADGROUP GROUP g1 c1 COUNT 1  STREAMS stream.oredes 0
                    // 指定队列名称，组名称，消费者名称，读取模式，读取数量，阻塞时间，队列名称，读取位置
                    List<MapRecord<String, Object, Object>> list = stringRedisTemplate.opsForStream().read(
                            Consumer.from("g1", "c1"),
                            StreamReadOptions.empty().count(1),
                            StreamOffset.create(queueName, ReadOffset.from("0"))
                    );
                    //2. 判断消息获取是否成功
                    if (list == null || list.isEmpty()) {
                        //2.1 获取失败 说明Pending List没有消息 ---->结束循环
                        break;
                    }

                    // 解析消息中的订单信息
                    MapRecord<String, Object, Object> record = list.get(0);
                    //  获取键值对集合
                    Map<Object, Object> values = record.getValue();
                    // 获取订单信息
                    VoucherOrder voucherOrder = BeanUtil.fillBeanWithMap(values, new VoucherOrder(), true);

                    //3. 获取成功，执行订单创建
                    handleVocherOrder(voucherOrder);

                    //4. ACK确认  SACK stream.orders g1 id
                    stringRedisTemplate.opsForStream().acknowledge(queueName, "g1", record.getId());

                } catch (Exception e) {
                    log.error("Pending List订单处理出现异常", e);
                    try {
                        Thread.sleep(20);
                    }catch (InterruptedException interruptedException){
                        interruptedException.printStackTrace();
                    }
                }
            }
        }
    }

    // 4. 取到了订单—————创建订单
    private void handleVocherOrder(VoucherOrder voucherOrder){
        // 获取用户
        Long userId = voucherOrder.getUserId();

        // 1. 创建锁对象
        RLock lock =  redissonClient.getLock("lock:order:" + userId);

        //2. 尝试获取锁
        boolean isLock = lock.tryLock();

        // 3. 判断锁是否获取成功
        if(! isLock){
            log.error("不允许重复下单");
            return;
        }
        try {
            proxy.createVoucherOrder(voucherOrder);
        } finally {
            // 4. 释放锁
            lock.unlock();
        }
    }

    /**
     *  秒杀优惠券下单------秒杀优化代码----lua脚本---主线程---使用Redis stream的消息队列完成的
     */
    private IVoucherOrderService proxy;
    @Override
    public Result seckillVoucher(Long voucherId) {
        // 获取用户
        Long userId = UserHolder.getUser().getId();
        // 获取订单id
        long orderId =  redisIdWorker.nextId("order");

        //1.执行Lua脚本
        Long result = stringRedisTemplate.execute(
                SECKILL_SCRIPT,
                Collections.emptyList(),
                voucherId.toString(),userId.toString(),String.valueOf(orderId)
        );
        //2.判断结果是否为0
        int r = result.intValue();
        if(r != 0){
            //3.不为0,代表没有购买资格
            return Result.fail(r == 1 ? "库存不足!" : "不能重复下单!");
        }

        //提前 获取代理对象
        proxy = (IVoucherOrderService) AopContext.currentProxy();

        //5.返回订单id
        return Result.ok(orderId);
    }


    /**
     * 秒杀优惠券下单------秒杀优化代码----创建订单
     * @param voucherOrder
     */
    @Transactional
    public void createVoucherOrder(VoucherOrder voucherOrder) {
        //4. 限制一人一单【悲观锁方案】
        Long userId = voucherOrder.getUserId();

        //4.1 查询订单
        int count = query().eq("user_id", userId).eq("voucher_id", voucherOrder.getVoucherId()).count();
        //4.2 判断订单是否存在
        // 是 -----> 返回异常信息---->结束
        if (count > 0) {
            log.error("用户已经购买了一次了");
        }

        //5. 扣减库存——解决超卖问题【乐观锁方案】
        boolean success = seckillVoucherService.update()
                .setSql("stock = stock-1")
                .eq("voucher_id", voucherOrder.getVoucherId()).gt("stock", 0) // 库存大于0就行了
                .update();
        if (!success) {
            log.error("库存不足");
        }

        //6. 创建订单
        save(voucherOrder);
    }
}

/*-----------------------------第三种方案结束-------------------------------------------------------------*/



